import SimpleITK as sitk
import matplotlib.pyplot as plt


# 单独写了一个用于显示nii图像切片的函数
def look_nii():
    image_path = 'C:/SYBdataset/dataset/qutougu/brain/s001.nii.gz'  # 已去头骨数据路径
    img = sitk.ReadImage(image_path)
    image = sitk.GetArrayFromImage(img)
    # 使用 matplotlib 瞄一眼切片图像
    for j in range(100, 140):
        slice_1 = image[:, j, :]
        # 在第一个轴上显示 slice_1 并添加颜色条
        im1 = plt.imshow(slice_1, cmap='gray')  # 使用灰度映射显示图片
        cbar1 = plt.colorbar(im1)  # 加个灰度的颜色条
        plt.show()
